# -*- coding: utf-8 -*-
"""
@Time    : 2024/7/19 11:56 
@Author  : ZhangShenao 
@File    : 6.使用智谱AI的Embedding模型.py 
@Desc    : 使用ZhipuAIEmbeddings,智谱AI的Embedding模型
"""
import dotenv
from langchain_community.embeddings import ZhipuAIEmbeddings

dotenv.load_dotenv()

# 创建ZhipuAIEmbeddings
# 使用embedding-2这个Embedding模型
embeddings = ZhipuAIEmbeddings(model='embedding-2')

# 执行Embedding
vector = embeddings.embed_documents(texts=['床前明月光'])[0]
print(vector[:10])
print(len(vector))
